import axios, { AxiosRequestConfig, Canceler } from 'axios';
import qs from 'qs';

import { isFunction } from '@/utils/is';

// 声明一个 Map 用于存储每个请求的标识和取消函数
let pendingMap = new Map<string, Canceler>();

/**
 * 获取请求的唯一标识
 * @param {AxiosRequestConfig} config 请求配置
 * @returns {string} 请求的唯一标识
 */
export const getPendingUrl = (config: AxiosRequestConfig): string => {
  // 使用请求方法、URL、请求参数和查询参数的字符串形式来生成唯一标识
  return [config.method, config.url, qs.stringify(config.data), qs.stringify(config.params)].join(
    '&',
  );
};

export class AxiosCanceler {
  /**
   * 添加请求到 pendingMap
   * @param {AxiosRequestConfig} config 请求配置
   */
  addPending(config: AxiosRequestConfig) {
    this.removePending(config); // 添加请求之前先移除已存在的相同请求
    const url = getPendingUrl(config);
    config.cancelToken =
      config.cancelToken ||
      new axios.CancelToken((cancel) => {
        if (!pendingMap.has(url)) {
          // 如果 pendingMap 中不存在当前请求，则添加进去
          pendingMap.set(url, cancel);
        }
      });
  }

  /**
   * 清空所有的 pending 请求
   */
  removeAllPending(): void {
    pendingMap.forEach((cancel) => {
      cancel && isFunction(cancel) && cancel();
    });
    pendingMap.clear();
  }

  /**
   * 移除请求从 pendingMap
   * @param {AxiosRequestConfig} config 请求配置
   */
  removePending(config: AxiosRequestConfig): void {
    const url = getPendingUrl(config);

    if (pendingMap.has(url)) {
      // 如果在 pendingMap 中存在当前请求标识，需要取消当前请求，并且移除
      const cancel = pendingMap.get(url);
      cancel && cancel(url);
      pendingMap.delete(url);
    }
  }

  /**
   * 重置 pendingMap
   */
  reset(): void {
    pendingMap = new Map<string, Canceler>();
  }
}
